﻿using System.Collections.Generic;

namespace BOL.Apps.GeneNetworkBrowser.Networks
{
    public static class TestNetwork
    {
        public static Network GetNetwork()
        {
            var network = new Network("Test", "This is a test network.", true);
            var fam138E = new Gene(100124412);
            var gh1 = new Gene(2688);
            var gh2 = new Gene(2689);
            var csh1 = new Gene(1442);
            var csh2 = new Gene(1443);
            var cshl1 = new Gene(1444);
            var genes = new HashSet<Gene> {fam138E, gh1, gh2, csh1, csh2, cshl1};

            var relation1 = new Relation(fam138E, gh1, 1);
            var relation2 = new Relation(fam138E, gh2, 1);
            var relation3 = new Relation(fam138E, csh1, 1);
            var relation4 = new Relation(fam138E, csh2, 1);
            var relation5 = new Relation(fam138E, cshl1, 1);
            var relation6 = new Relation(csh1, csh2, 1);
            var relation7 = new Relation(csh1, cshl1, 1);
            var relations = new HashSet<Relation>();
            relations.Add(relation1);
            relations.Add(relation2);
            relations.Add(relation3);
            relations.Add(relation4);
            relations.Add(relation5);
            relations.Add(relation6);
            relations.Add(relation7);

            network.AddVertexRange(genes);
            network.AddEdgeRange(relations);

            return network;
        }
    }
}
